# -*- coding: utf-8 -*-
# @Time    : 2020/7/1 下午11:50
# @Author  : lilong

"""
以最小的数据训练机器人的nlu部分
"""

from rasa.train import train_nlu

config = "../bot_playground/bot_tiny/config.yml"
train_files = "../bot_playground/bot_tiny/data"
output = "../bot_playground/bot_tiny/models/nlu"
train_path = "../bot_playground/tmp_dir"

train_path = train_nlu(config, train_files, output, persist_nlu_training_data=True)
# train_path = train_nlu(config, train_files, output)

print("train_path:", train_path)